Node / ReticulumProjects / MeshChatX.git / files / tests / backend / test_propagation_nodes_robustness.py

Displaying Raw • Download

Tff7b72import T7ee787json

Tff7b72import T7ee787shutil

Tff7b72import T7ee787tempfile

Tff7b72from T7ee787unittestT7ee787.T7ee787mock Tff7b72import Te6edf3MagicMockTb4b4b4, Te6edf3patch

Tff7b72import T7ee787pytest

Tff7b72import T7ee787RNS

Tff7b72from T7ee787meshchatxT7ee787.T7ee787meshchat Tff7b72import Te6edf3ReticulumMeshChat

Tf0883e@pytestTff7b72.Td2a8fffixture

Tff7b72def Td2a8fftemp_dirTb4b4b4(Tb4b4b4)Tb4b4b4:

Te6edf3dir_path Tff7b72= Te6edf3tempfileTff7b72.Td2a8ffmkdtempTb4b4b4(Tb4b4b4)

Tff7b72yield Te6edf3dir_path

Te6edf3shutilTff7b72.Td2a8ffrmtreeTb4b4b4(Te6edf3dir_pathTb4b4b4)

Tf0883e@pytestTff7b72.Td2a8fffixture

Tff7b72def Td2a8ffmock_rns_minimalTb4b4b4(Tb4b4b4)Tb4b4b4:

Tff7b72with Tb4b4b4(

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.ReticulumTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_rnsTb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.TransportTa5d6ff"Tb4b4b4)Tb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffLXMF.LXMRouterTa5d6ff"Tb4b4b4)Tb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.meshchat.get_file_pathTa5d6ff"Tb4b4b4, Te6edf3return_valueTff7b72=Ta5d6ff"Ta5d6ff/tmp/mock_pathTa5d6ff"Tb4b4b4)Tb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.meshchat.generate_ssl_certificateTa5d6ff"Tb4b4b4)Tb4b4b4,

Tb4b4b4)Tb4b4b4:

Te6edf3mock_rns_instance Tff7b72= Te6edf3mock_rnsTff7b72.Td2a8ffreturn_value

Te6edf3mock_rns_instanceTff7b72.Td2a8ffconfigpath Tff7b72= Ta5d6ff"Ta5d6ff/tmp/mock_configTa5d6ff"

Te6edf3mock_rns_instanceTff7b72.Td2a8ffis_connected_to_shared_instance Tff7b72= Tff7b72False

T8b949e# Use a real identity to satisfy RNS.Destination.hash

Te6edf3real_id Tff7b72= Te6edf3RNSTff7b72.Td2a8ffIdentityTb4b4b4(Tb4b4b4)

T8b949e# We can still mock parts of it if needed, but RNS expects

T8b949e# isinstance(identity, RNS.Identity) to pass or bytes of specific length

Tff7b72yield Te6edf3real_id

Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio

Tff7b72async Tff7b72def Td2a8fftest_propagation_nodes_endpoint_robustnessTb4b4b4(Te6edf3mock_rns_minimalTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Te6edf3app_instance Tff7b72= Te6edf3ReticulumMeshChatTb4b4b4(

Te6edf3identityTff7b72=Te6edf3mock_rns_minimalTb4b4b4,

Te6edf3storage_dirTff7b72=Te6edf3temp_dirTb4b4b4,

Te6edf3reticulum_config_dirTff7b72=Te6edf3temp_dirTb4b4b4,

Tb4b4b4)

T8b949e# Find the propagation nodes handler

Te6edf3handler Tff7b72= Tff7b72None

Tff7b72for Te6edf3route Tff7b72in Te6edf3app_instanceTff7b72.Td2a8ffget_routesTb4b4b4(Tb4b4b4)Tb4b4b4:

Tff7b72if Te6edf3routeTff7b72.Td2a8ffpath Tff7b72== Ta5d6ff"Ta5d6ff/api/v1/lxmf/propagation-nodesTa5d6ff" Tff7b72and Te6edf3routeTff7b72.Td2a8ffmethod Tff7b72== Ta5d6ff"Ta5d6ffGETTa5d6ff"Tb4b4b4:

Te6edf3handler Tff7b72= Te6edf3routeTff7b72.Td2a8ffhandler

Tff7b72break

Tff7b72assert Te6edf3handler Tff7b72is Tff7b72not Tff7b72None

T8b949e# Test with valid limit

Te6edf3request Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3requestTff7b72.Td2a8ffquery Tff7b72= Tb4b4b4{Ta5d6ff"Ta5d6fflimitTa5d6ff"Tb4b4b4: Ta5d6ff"Ta5d6ff10Ta5d6ff"Tb4b4b4}

Te6edf3response Tff7b72= Tff7b72await Te6edf3handlerTb4b4b4(Te6edf3requestTb4b4b4)

Tff7b72assert Te6edf3responseTff7b72.Td2a8ffstatus Tff7b72== T79c0ff200

Te6edf3data Tff7b72= Te6edf3jsonTff7b72.Td2a8ffloadsTb4b4b4(Te6edf3responseTff7b72.Td2a8ffbodyTb4b4b4)

Tff7b72assert Ta5d6ff"Ta5d6fflxmf_propagation_nodesTa5d6ff" Tff7b72in Te6edf3data

T8b949e# Test with invalid limit (should not crash)

Te6edf3requestTff7b72.Td2a8ffquery Tff7b72= Tb4b4b4{Ta5d6ff"Ta5d6fflimitTa5d6ff"Tb4b4b4: Ta5d6ff"Ta5d6ffinvalidTa5d6ff"Tb4b4b4}

Te6edf3response Tff7b72= Tff7b72await Te6edf3handlerTb4b4b4(Te6edf3requestTb4b4b4)

Tff7b72assert Te6edf3responseTff7b72.Td2a8ffstatus Tff7b72== T79c0ff200

Te6edf3data Tff7b72= Te6edf3jsonTff7b72.Td2a8ffloadsTb4b4b4(Te6edf3responseTff7b72.Td2a8ffbodyTb4b4b4)

Tff7b72assert Ta5d6ff"Ta5d6fflxmf_propagation_nodesTa5d6ff" Tff7b72in Te6edf3data

T8b949e# Test with missing limit (should not crash)

Te6edf3requestTff7b72.Td2a8ffquery Tff7b72= Tb4b4b4{Tb4b4b4}

Te6edf3response Tff7b72= Tff7b72await Te6edf3handlerTb4b4b4(Te6edf3requestTb4b4b4)

Tff7b72assert Te6edf3responseTff7b72.Td2a8ffstatus Tff7b72== T79c0ff200

Te6edf3data Tff7b72= Te6edf3jsonTff7b72.Td2a8ffloadsTb4b4b4(Te6edf3responseTff7b72.Td2a8ffbodyTb4b4b4)

Tff7b72assert Ta5d6ff"Ta5d6fflxmf_propagation_nodesTa5d6ff" Tff7b72in Te6edf3data